python3 tools/train_net.py --config-file ./configs/bee/sbs_S50.yml MODEL.DEVICE "cuda:0"